"""
使用 LangServe 部署通义大模型应用
"""

from fastapi import FastAPI
from langserve import add_routes
from main import create_app

# 创建 FastAPI 应用
app = FastAPI(
    title="通义大模型 API",
    version="1.0",
    description="基于通义大模型的 AI 助手 API"
)

# 创建 LLM 应用实例
llm_app = create_app(debug=False)

# 添加 LangServe 路由
add_routes(
    app,
    llm_app.get_chain(),
    config_keys=["configurable"],
    path="/chat"
)

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000) 